import pandas as pd
import numpy as np

# 创建示例数据
data = {
    '姓名': ['张三', '李四', '王五', '赵六', '钱七'],
    '年龄': [25, 30, 35, 28, 32],
    '城市': ['北京', '上海', '广州', '深圳', '杭州'],
    '薪资': [8000, 12000, 15000, 10000, 13000],
    '部门': ['技术', '销售', '技术', '人事', '技术']
}

df = pd.DataFrame(data)
print("员工数据:")
print(df)

# 选择列
print("\n选择'姓名'列:")
print(df['姓名'])

print("\n选择多列:")
print(df[['姓名', '薪资']])

# 选择行
print("\n选择第一行:")
print(df.iloc[0])  # 位置索引

print("\n选择前3行:")
print(df.iloc[0:3])

print("\n根据条件选择行:")
print(df[df['薪资'] > 10000])

# 添加新列
df['奖金'] = df['薪资'] * 0.1
print("\n添加奖金列后:")
print(df)

# 修改数据
df.loc[df['部门'] == '技术', '薪资'] += 2000
print("\n技术部门加薪后:")
print(df)